iT邦幫忙

2021 iThome 鐵人賽

DAY 10
0
自我挑戰組

IOS、Python自學心得30天系列 第 10

IOS、Python自學心得30天 Day-10 模組訓練改善-2

  • 分享至 

  • xImage
  •  

前言:
儲存上次進度訓練模組後,目前遇到一個小問題,載入方面一直沒試成功,所以我先做了一個辨別模型的if,以便於之後針對各種狀況切換訓練

程式碼:

if os.path.exists(DATASET_PATH):
    if os.path.exists(DATASET_PATH + WEIGHTS_FINAL):
        print(WEIGHTS_FINAL + "模型存在,將繼續訓練模型")
    else:
        print(WEIGHTS_FINAL + '模型不存在,將新建訓練模型')
        # 訓練模型
        net_final.fit(train_batches,
                      steps_per_epoch = train_batches.samples // BATCH_SIZE,
                      validation_data = valid_batches,
                      validation_steps = valid_batches.samples // BATCH_SIZE,
                      epochs = NUM_EPOCHS,
                      callbacks = [cp_callback])
else:
    print(WEIGHTS_FINAL + '路徑不存在,請確認路徑')
      
# 儲存訓練好的模型
print("儲存訓練模型")
net_final.save(WEIGHTS_FINAL)

上一篇
IOS、Python自學心得30天 Day-9 模組訓練改善-1
下一篇
IOS、Python自學心得30天 Day-11 模組訓練改善-3
系列文
IOS、Python自學心得30天30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言